Spiral Pattern in python